from config import to_abs_path


class FasttextConfig:
    def __init__(self):
        self.train_path = to_abs_path('data/fasttext-yt/train.txt')
        self.test_path = to_abs_path('data/fasttext-yt/test.txt')
        self.val_path = to_abs_path('data/fasttext-yt/val.txt')
        self.compare_data_dir = to_abs_path('data/fasttext-yt/test/')
        self.model_path = to_abs_path('model/fasttext-yt/fasttext.bin')
        self.jieba_allow_pos = {'n', 'nr', 'ns', 'nt', 'nw', 'nz', 'v', 'vn', 'a'}
        self.thulac_allow_pos = {'n', 'np', 'ns', 'ni', 'nz', 'v', 'a'}

        self.processed_train_path = to_abs_path('data/fasttext-yt/processed_train.txt')
        self.processed_test_path = to_abs_path('data/fasttext-yt/processed_test.txt')
        self.processed_val_path = to_abs_path('data/fasttext-yt/processed_val.txt')


f_conf = FasttextConfig()
